import torch
from ..data.mnist import mnist_train, mnist_eval

print(mnist_train)